python program for armstrong number